package com.ahui.ydd.controller;

import com.ahui.ydd.common.CustomException;
import com.ahui.ydd.common.R;
import com.ahui.ydd.dto.PageDto;
import com.ahui.ydd.entity.YddLink;
import com.ahui.ydd.entity.YddUser;
import com.ahui.ydd.service.YddUserService;
import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
import com.baomidou.mybatisplus.extension.plugins.pagination.Page;
import org.apache.commons.lang.StringUtils;
import org.checkerframework.checker.units.qual.A;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.util.DigestUtils;
import org.springframework.web.bind.annotation.*;

import java.io.UnsupportedEncodingException;
import java.nio.charset.StandardCharsets;
import java.util.ArrayList;
import java.util.List;

@RestController
@RequestMapping("/ydd_user")
public class YddUserController {

    @Autowired
    private YddUserService yddUserService;
    /**
     * 分页查询
     * @param sex
     * @param username
     * @param mobile
     * @param page
     * @param limit
     * @return
     */
    @GetMapping("/list")
    public R<PageDto> page(String sex, String username, String mobile, Integer page, Integer limit) {



        //构造分页构造器对象
        Page<YddUser> pageInfo = new Page<>(page, limit);
        //条件构造器
        LambdaQueryWrapper<YddUser> queryWrapper = new LambdaQueryWrapper<>();
        //添加过滤条件
        queryWrapper.like(StringUtils.isNotEmpty(username),YddUser::getUsername,username);
        queryWrapper.eq( StringUtils.isNotEmpty(sex),YddUser::getSex,sex);
        queryWrapper.like( StringUtils.isNotEmpty(mobile),YddUser::getMobile,mobile);
        //添加排序条件
        queryWrapper.orderByAsc(YddUser::getId);
        //执行分页查询
       yddUserService.page(pageInfo, queryWrapper);
//        List<YddUser> UserList = pageInfo.getRecords();
//        if ("1".equals(sex)) {
//            for (YddUser yddUser : UserList) {
//                yddUser.setSex("男");
//            }
//        } else if ("2".equals(sex)) {
//            for (YddUser yddUser  : UserList) {
//                yddUser.setSex("女");
//            }
//
//        }
//        pageInfo.setRecords(UserList);
        List<YddUser> userList = pageInfo.getRecords();
        PageDto pageDto = new PageDto();
        pageDto.setPageSize(limit);
        pageDto.setCurrentPage(page);
        pageDto.setData(userList);
        pageDto.setCount((int)pageInfo.getTotal());
        pageDto.setTotalPages((int)pageInfo.getPages());

        return R.success(pageDto);
    }

    /**
     * 根据id查询 修改01 数据回显
     *
     * @param id
     * @return
     */
    @GetMapping("/rowInfo")
    public R<YddUser> getById(String id) {

        LambdaQueryWrapper<YddUser> queryWrapper = new LambdaQueryWrapper<>();
        queryWrapper.eq(YddUser::getId, id);
        YddUser user   = yddUserService.getOne(queryWrapper);
        //  = yddLinkService.getById(id);
        return R.success(user);

    }

    /**
     * 修该02
     * @param
     * @return
     */
    @PutMapping("/update")
    public R<String> update(@RequestBody YddUser yddUser) {
        //密码加密
        String password =yddUser.getPassword();
        String pass = DigestUtils.md5DigestAsHex(password.getBytes());
        yddUser.setPassword(pass);
        yddUserService.updateById(yddUser);
        return R.success("修改成功");
    }

    /**
     * 删除链接
     * @param id
     * @return
     */
    @DeleteMapping("/del")
    public R<String> delete(String id){
        yddUserService.removeById(id);
        return R.success("删除成功");
    }

    /**
     * 添加链接
     * @param
     * @return
     */

    @PostMapping("/add")
    public R<String> add(@RequestBody YddUser yddUser){
        //密码加密
        String password =yddUser.getPassword();
        String pass = DigestUtils.md5DigestAsHex(password.getBytes());
        yddUser.setPassword(pass);
        //根据手机号判断用户是否存在
        LambdaQueryWrapper<YddUser> queryWrapper = new LambdaQueryWrapper<>();
        queryWrapper.eq(YddUser::getMobile,yddUser.getMobile());
        YddUser user = yddUserService.getOne(queryWrapper);
        if(user!=null){
            throw  new CustomException("该用户已存在");
        }
        yddUserService.saveOrUpdate(yddUser);
        return R.success("添加成功");
    }
}
